package org.dromara.common.mybatis.helper;

import cn.hutool.core.convert.Convert;
import com.baomidou.dynamic.datasource.DynamicRoutingDataSource;
import lombok.AccessLevel;
import lombok.NoArgsConstructor;
import org.dromara.common.core.exception.ServiceException;
import org.dromara.common.core.utils.SpringUtils;
import org.dromara.common.mybatis.enums.DataBaseType;

import javax.sql.DataSource;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;


/**
 * @Author 刘武贵
 * @Date 2024/8/15 22:00
 * @Description 功能描述：数据库助手
 */
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public class DataBaseHelper {

    /**
     * 定义一个动态路由数据源对象，用于在不同的数据源之间动态切换
     */
    private static final DynamicRoutingDataSource DS = SpringUtils.getBean(DynamicRoutingDataSource.class);

    /**
     * 获取当前数据库类型
     * 此方法用于确定当前使用的数据库类型。它通过数据源获取数据库连接，
     * 然后查询数据库的元数据信息来识别数据库类型。
     *
     * @return 返回识别出的数据库类型
     * @throws ServiceException 如果无法获取数据库连接或元数据信息，则抛出服务异常
     */
    public static DataBaseType getDataBaseType() {
        // 根据当前事务或上下文确定合适的数据源
        DataSource dataSource = DS.determineDataSource();
        try (Connection conn = dataSource.getConnection()) {
            // 获取数据库的元数据
            DatabaseMetaData metaData = conn.getMetaData();
            // 获取数据库产品名称
            String databaseProductName = metaData.getDatabaseProductName();
            // 根据数据库产品名称查找并返回对应的数据库类型
            return DataBaseType.find(databaseProductName);
        } catch (SQLException e) {
            // 如果在获取数据库连接或元数据信息过程中发生错误，包装并抛出服务异常
            throw new ServiceException(e.getMessage());
        }
    }

    public static boolean isMySql() {
        return DataBaseType.MY_SQL == getDataBaseType();
    }

    public static boolean isOracle() {
        return DataBaseType.ORACLE == getDataBaseType();
    }

    public static boolean isPostgerSql() {
        return DataBaseType.POSTGRE_SQL == getDataBaseType();
    }

    public static boolean isSqlServer() {
        return DataBaseType.SQL_SERVER == getDataBaseType();
    }

    /**
     * 在给定的字符串集合中查找指定的元素。
     *
     * @param var1 要查找的元素，将被转换为字符串进行比较。
     * @param var2 字符串集合，格式为逗号分隔的值。
     * @return 返回一个字符串，包含用于不同数据库类型的查询条件，用于判断var1是否存在于var2中。
     */
    public static String findInSet(Object var1, String var2) {
        // 获取当前数据库的类型，以便根据不同的数据库生成相应的查询语句
        DataBaseType dataBasyType = getDataBaseType();

        // 将输入的元素转换为字符串，以便进行查询
        String var = Convert.toStr(var1);

        // 根据不同的数据库类型生成相应的查询语句
        if (dataBasyType == DataBaseType.SQL_SERVER) {
            // charindex(',100,' , ',0,100,101,') <> 0 SQL Server中使用charindex函数查询，将待查询元素和目标集合都用逗号包围后进行匹配
            return "charindex(',%s,' , ','+%s+',') <> 0".formatted(var, var2);
        } else if (dataBasyType == DataBaseType.POSTGRE_SQL) {
            // (select position(',100,' in ',0,100,101,')) <> 0 PostgreSQL中使用position函数查询，将待查询元素和目标集合都用逗号和双引号包围后进行匹配
            return "(select position(',%s,' in ','||%s||',')) <> 0".formatted(var, var2);
        } else if (dataBasyType == DataBaseType.ORACLE) {
            // instr(',0,100,101,' , ',100,') <> 0 Oracle中使用instr函数查询，将目标集合和待查询元素都用逗号包围后进行匹配
            return "instr(','||%s||',' , ',%s,') <> 0".formatted(var2, var);
        }
        // find_in_set(100 , '0,100,101') 默认情况下，使用标准的SQL find_in_set函数查询，将待查询元素和目标集合都用逗号包围后进行匹配
        return "find_in_set('%s' , %s) <> 0".formatted(var, var2);
    }

    /**
     * 获取当前加载的数据库名
     */
    public static List<String> getDataSourceNameList() {
        return new ArrayList<>(DS.getDataSources().keySet());
    }
}
